import requests
from api.ihrm_login_api import IhrmLoginApi


class IhrmAddApi:

    @classmethod
    def add(cls, req_body):
        # 准备登录请求的 体
        req_body_l = {"mobile": "13800000002", "password": "123456"}
        # 调用接口层的 登录请求类的方法 参数中传递请求体，获取响应数据
        res = IhrmLoginApi.login(req_body_l)
        # 整理成完整的令牌
        token = "Bearer " + res.json()["data"]

        # 准备员工添加接口的请求行头体
        req_url = "http://ihrm-test.itheima.net/api/sys/user"
        req_head = {"Authorization": token}

        # 发送请求，获取响应数据
        res = requests.post(url=req_url, headers=req_head, json=req_body)
        print("响应数据===", res)
        return res
